from txl_data import read_txl,write_txl
import os

txl=read_txl()#读取通讯录.txt文件,将通讯录内容存为列表

for i in range(10000):
    #显示菜单
    print("请输入编号选择功能:")
    print("1.浏览通讯录")
    print("2.按人名查找并显示")
    print("3.添加联系人")
    print("4.删除联系人")
    print("5.修改联系人")
    print("6.保存")
    choose=input("请选择:")
    #1.浏览通讯录
    print()
    if choose=="1":
        print("   姓名\t\t电话号码\t班级")
        for i in range(0,len(txl),3):
            print(f"{(i//3)+1}: {txl[i]} \t{txl[i+1]} \t{txl[i+2]}")

    #2.按人名查找并显示
    if choose=="2":
        name=input("请输入带查找的联系人姓名:")
        print("   姓名\t\t电话号码\t班级")
        for i in range(0,len(txl),3):
            if txl[i]==name:
                print(f"{(i//3)+1}: {txl[i]} \t{txl[i+1]} \t{txl[i+2]}")

    #3.添加联系人
    if choose=="3":
        name=input("联系人姓名:")
        tel=input("联系人电话:")
        classroom=input("联系人班级:")
        txl.append(name)
        txl.append(tel)
        txl.append(classroom)
        print("   姓名\t\t电话号码\t班级")
        print(f"{(len(txl)//3)+1}: {txl[-3]} \t{txl[-2]} \t{txl[-1]}")

    #4.删除联系人
    if choose=="4":
        name=input("请输入待删除的联系人姓名:")
        del_num=0
        for i in range(len(txl)-3,-1,-3):
            if txl[i]==name:
                del txl[i+2]
                del txl[i+1]
                del txl[i]
                del_num+=1
        print(f"删除{del_num}条记录")

    #5.修改联系人
    if choose=="5":
        id=int(input("请输入待修改的联系人编号:"))
        print("   姓名\t\t电话号码\t班级")
        print(f"{id}: {id*3-3} \t{txl[id*3-2]} \t{txl[id*3-1]}")
        print()
        name=input("修改后联系人姓名:")
        tel=input("修改后联系人电话:")
        classroom=input("修改后联系人班级:")  
        txl[id*3-3]=name      
        txl[id*3-2]=tel      
        txl[id*3-1]=classroom      
        print("   姓名\t\t电话号码\t班级")
        print(f"{id}: {txl[id*3-3]} \t{txl[id*3-2]} \t{txl[id*3-1]}")
    
    #保存
    if choose=="6":
        write_txl(txl)#将修改后的列表写入通讯录
        print("保存成功！")

    print()
    input("按任意键继续……")
    os.system('cls' if os.name == 'nt' else 'clear')

